package main

/*
Go实现的支持PostgreSQL的驱动也很多，因为国外很多人在开发中使用了这个数据库。

https://github.com/lib/pq 支持database/sql驱动，纯Go写的
https://github.com/jbarham/gopgsqldriver 支持database/sql驱动，纯Go写的
https://github.com/lxn/go-pgsql 支持database/sql驱动，纯Go写的
在下面的示例中我采用了第一个驱动，因为它目前使用的人最多

还有pg不支持LastInsertId函数，因为PostgreSQL内部没有实现类似MySQL的自增ID返回，其他的代码几乎是一模一样。
*/

import (
	"database/sql"
	"fmt"
	_ "github.com/lib/pq"
	"time"
)

func main() {
	db, err := sql.Open("postgres", "user=postgres password=1234 dbname=test sslmode=disable")
	checkErr(err)

	// 插入数据
	stmt, err := db.Prepare("INSERT INTO userinfo(username,department,created) VALUES($1,$2,$3) RETURNING uid")
	checkErr(err)

	res, err := stmt.Exec("dralee", "研发部门", time.Now())
	checkErr(err)

	// pg不支持res.LastInsertId函数，因为他没有类似MYSQL的睡在增Id

	var lastInsertId int
	err = db.QueryRow("INSERT INTO userinfo(username,department,created) VALUES ($1,$2,$3) RETURNING uid;", "dralee", "研发部门", time.Now()).Scan(&lastInsertId)
	checkErr(err)
	fmt.Println("最后插入id = ", lastInsertId)

	// 更新数据
	stmt, err = db.Prepare("UPDATE userinfo SET username=$1 WHERE uid=$2")
	checkErr(err)
	res, err = stmt.Exec("dralee2", lastInsertId)
	checkErr(err)
	affect, err := res.RowsAffected()
	checkErr(err)

	fmt.Println("影响行数：", affect)

	// 查询数据
	rows, err := db.Query("SELECT * FROM userinfo")
	checkErr(err)

	for rows.Next() {
		var uid int
		var username string
		var department string
		var created string
		err = rows.Scan(&uid, &username, &department, &created)
		checkErr(err)
		fmt.Println("uid: ", uid)
		fmt.Println("username: ", username)
		fmt.Println("department: ", department)
		fmt.Println("created: ", created)
	}

	// 删除数据
	stmt, err = db.Prepare("DELETE FROM userinfo WHERE uid=$1")
	checkErr(err)

	res, err = stmt.Exec(lastInsertId)
	checkErr(err)

	affect, err = res.RowsAffected()
	checkErr(err)

	fmt.Println("删除影响行数：", affect)

	db.Close()
}

func checkErr(err error) {
	if err != nil {
		panic(err)
	}
}
